import torch

A = torch.arange(24).view(2, 3, 4)
print(A)
B = A.permute(0, 2, 1)
print(B)
C = A.transpose(1, 2)
D = A.swapaxes(1, 2)
print(C)
print(D)
print(A[0][0][1])